package pl.diagnoser.server.persistance.dao;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import org.hibernate.Session;
import org.hibernate.criterion.Expression;
import pl.diagnoser.client.dto.AttributeValuesDictDTO;
import pl.diagnoser.server.persistance.map.AttributeDict;
import pl.diagnoser.server.persistance.map.AttributeValuesDict;
import pl.diagnoser.server.persistance.tools.HibernateUtil;

public class AttributeValuesDictDao implements Serializable {

	private static final long serialVersionUID = 1L;
	
	@SuppressWarnings("unchecked")
	public List<AttributeValuesDictDTO> getAttributeValuesList() {
		Session session = HibernateUtil.getSessionFactory().openSession();
		session.beginTransaction();
		List<AttributeValuesDict> attributeValuesList = new ArrayList<AttributeValuesDict>(session.createQuery("from AttributeValuesDict").list());
		List<AttributeValuesDictDTO> attributeValuesListDTO = new ArrayList<AttributeValuesDictDTO>(attributeValuesList != null ? attributeValuesList.size() : 0);
		
		if(attributeValuesList != null) {
			for (AttributeValuesDict attributeValues : attributeValuesList) {
				attributeValuesListDTO.add(createAttributeValuesDictDTO(attributeValues));
			}
		}
		session.getTransaction().commit();
		session.close();
		return attributeValuesListDTO;
	}
	
	public String saveAttributeValue(String attributeValue, String attribute) {
		String ret = "OK";
		Session session = HibernateUtil.getSessionFactory().openSession();
		session.beginTransaction();
	    try {
	    	AttributeDict tmpAttribute = (AttributeDict) session.createCriteria(AttributeDict.class).add(Expression.eq("attributeName", attribute)).uniqueResult();
	    	tmpAttribute.getAttributeValuesDict().add(new AttributeValuesDict(attributeValue));
	    	session.save(tmpAttribute);
	    	session.getTransaction().commit();
	    } catch(Exception e) {
	    	ret = e.getMessage();
	    } finally {
			session.close();
	    }
		return ret;
	}
	
	public String removeAttributeValue(String attributeValue, String attribute) {
		String ret = "OK";
		Session session = HibernateUtil.getSessionFactory().openSession();
		session.beginTransaction();
	    try {
	    	AttributeDict tmpAttribute = (AttributeDict) session.createCriteria(AttributeDict.class).add(Expression.eq("attributeName", attribute)).uniqueResult();
	    	AttributeValuesDict tmpAttributeValue = null;
	    	for(AttributeValuesDict attributeValueTmp : tmpAttribute.getAttributeValuesDict()) {
	    		if(attributeValueTmp.getAttributeValue().equals(attributeValue))
	    			tmpAttributeValue = attributeValueTmp;
	    	}
	    	if(tmpAttributeValue != null)
	    		tmpAttribute.getAttributeValuesDict().remove(tmpAttributeValue);
	    	
	    	session.save(tmpAttribute);
	    	session.getTransaction().commit();
	    } catch(Exception e) {
	    	ret = e.getMessage();
	    } finally {
			session.close();
	    }   
		return ret;
	}
	
	public AttributeValuesDictDTO createAttributeValuesDictDTO( AttributeValuesDict attributeValuesDict ) {
		return new AttributeValuesDictDTO( attributeValuesDict.getAttributeValue() );
	}
}
